package mine.code.question.动态规划;

import org.junit.Test;

/**
 * 给你一个整数数组nums，你可以对它进行一些操作。
 * <p>
 * 每次操作中，选择任意一个nums[i]，删除它并获得nums[i]的点数。之后，你必须删除 所有 等于nums[i] - 1 和 nums[i] + 1的元素。
 * <p>
 * 开始你拥有 0 个点数。返回你能通过这些操作获得的最大点数。
 * <p>
 * 示例 1：
 * <p>
 * 输入：nums = [3,4,2]
 * 输出：6
 * 解释：
 * 删除 4 获得 4 个点数，因此 3 也被删除。
 * 之后，删除 2 获得 2 个点数。总共获得 6 个点数。
 *
 * @author caijinnan
 */
public class 删除并获得点数 {

    @Test
    public void run() {
        // 1 7 8 8 8 9
        int[] nums = {1, 7 ,8 ,8 ,8, 9};
        System.out.println(deleteAndEarn(nums));
    }


    public int deleteAndEarn(int[] nums) {
        int[] arr = new int[10001];
        for (int num : nums) {
            arr[num] += 1;
        }
        int[] dp = new int[10001];
        dp[0] = arr[0];
        dp[1] = arr[1];
        for (int i = 2; i <arr.length ; i++) {
            dp[i] = Math.max(dp[i - 1], dp[i - 2] + arr[i] * i);
        }
        return dp[10000];
    }
}
